package com.zcxy.fast.util;

import lombok.Data;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.List;

@Component
@Data
public class AutoUtils {
    public static List<String> getTemplates(String type) {
        List<String> templates = new ArrayList();
        //单表
        if("1".equals(type)){
            templates.add("commonJava/controller.java.vm");
            templates.add("commonJava/Entity.java.vm");
            templates.add("commonJava/EntityVo.java.vm");
            templates.add("commonJava/mapper.java.vm");
            templates.add("commonJava/service.java.vm");
            templates.add("commonJava/serviceImpl.java.vm");
            templates.add("simpleOneTable/api.js.vm");
            templates.add("simpleOneTable/index.vue.vm");
            templates.add("simpleOneTable/local.js.vm");
        }
        //树表
        else if("6".equals(type)){
            templates.add("commonJava/controller.java.vm");
            templates.add("commonJava/Entity.java.vm");
            templates.add("commonJava/EntityVo.java.vm");
            templates.add("commonJava/mapper.java.vm");
            templates.add("commonJava/service.java.vm");
            templates.add("commonJava/serviceImpl.java.vm");
            templates.add("simpleOneTable/api.js.vm");
            templates.add("simpleOneTable/tree.vue.vm");
            templates.add("simpleOneTable/local.js.vm");
        }
        //主表
        else if("4".equals(type)){
            templates.add("commonJava/controller.java.vm");
            templates.add("commonJava/Entity.java.vm");
            templates.add("commonJava/EntityVo.java.vm");
            templates.add("commonJava/mapper.java.vm");
            templates.add("commonJava/service.java.vm");
            templates.add("commonJava/serviceImpl.java.vm");
            templates.add("simpleOneTable/api.js.vm");
            templates.add("simpleOneTable/index.vue.vm");
            templates.add("simpleOneTable/local.js.vm");
        }
        //附表
        else if("5".equals(type)){
            templates.add("commonJava/controller.java.vm");
            templates.add("commonJava/Entity.java.vm");
            templates.add("commonJava/EntityVo.java.vm");
            templates.add("commonJava/mapper.java.vm");
            templates.add("commonJava/service.java.vm");
            templates.add("commonJava/serviceImpl.java.vm");
            templates.add("simpleOneTable/api.js.vm");
            templates.add("simpleOneTable/index.vue.vm");
            templates.add("simpleOneTable/local.js.vm");
        }

        return templates;
    }
}
